package leetcode.hot100;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

@SuppressWarnings("all")
public class _15_三数之和 {

    // 数组nums = [-1,0,1,2,-1,-4] 中满足三数之和为0的有 [-1,-1,2]和[-1,0,1]
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> resultList = new ArrayList<>();
        if (nums == null || nums.length == 0) return resultList;
        // [-4,-1,-1,0,1,2]
        Arrays.sort(nums);
        int i = 0;
        while (i < nums.length) {
            // 在固定当前数之前先判断之前是否已经定位过了，至少保证有三个数才有定位的意义
            while (i - 1 >= 0 && i < nums.length - 2 && nums[i] == nums[i - 1]) {
                i++;
            }
            // 固定一个数，找两数之和，为了去重，相同的数只能固定一次
            int target = nums[i];
            // 第二个数的位置
            int left = i + 1;
            // 第三个数的位置：最开始永远是数组末尾
            int right = nums.length - 1;
            // 固定了一个数，还得确保有两个数
            while (left < right) {
                int sum = nums[left] + nums[right] + target;
                if (sum == 0) {
                    // [-2,0,0,2,2] ---> [-2,0,2]
                    resultList.add(Arrays.asList(target, nums[left++], nums[right--]));
                    // 为了去重，已经固定的第二个数和第三个数也要跳过
                    while (left < right && nums[left] == nums[left - 1]) {
                        left++;
                    }
                    while (left < right && nums[right] == nums[right + 1]) {
                        right--;
                    }
                } else if (sum < 0) {
                    left++;
                } else {
                    right--;
                }
            }
            // 定位下一个数
            i++;
        }
        return resultList;
    }
}
